Skip to content

Add LIBERO-10 action-policy finetune cookbook#242

Merged
lfengad merged 9 commits into
NVIDIA:mainfrom
fwd4:haolia/add-action-policy-libero-finetune
Jul 2, 2026
Merged

Add LIBERO-10 action-policy finetune cookbook#242
lfengad merged 9 commits into
NVIDIA:mainfrom
fwd4:haolia/add-action-policy-libero-finetune

Conversation

@fwd4

@fwd4 fwd4 commented Jun 26, 2026

Copy link
Copy Markdown
Collaborator

What

Adds a LIBERO action-policy SFT cookbook, mirroring the existing DROID action-policy finetune cookbook.

To match the LIBERO-10 results reported in Cosmos3, it ships two presets (both lr 5e-5, warmup 500, cycle 16000, global batch 2048 / HSDP 2x8):

  • (A) libero_10-onlylaunch_sft_action_policy_libero.sh + toml/sft_config/action_policy_libero_repro.toml (max_iter 2000); stages nvidia/LIBERO_LeRobot_v3 libero_10.
  • (B) libero-all (4-suite equal mix)launch_sft_action_policy_libero_all.sh + toml/sft_config/action_policy_libero_all_repro.toml (max_iter 5000; needs longer to converge); stages all 4 suites.

Each launcher is a local wrapper: stages the dataset, downloads the Wan VAE, converts the Cosmos3-Nano base to DCP, and launches training. Extends finetune/README.md (recipe table + LIBERO quick start for both presets) and links it from the action README.md.

Delegates the canonical training implementation to cosmos-framework (companion PR #61 adds the action_policy_libero_nano / action_policy_libero_all_nano experiments + eval harness).

Mirrors the DROID cookbook structure. Based on main.

🤖 Generated with Claude Code

fwd4 and others added 8 commits June 26, 2026 21:05
Mirror the DROID action-policy cookbook for LIBERO-10: launch_sft_action_policy_libero.sh
+ action_policy_libero_repro.toml + finetune README section + action README link.
Stages nvidia/LIBERO_LeRobot_v3 libero_10, no keep-ranges filter, lr 5e-5/wu500/cyc16k.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
…only

Two presets to match the Cosmos3 LIBERO-10 result: (A) libero_10-only
(launch_sft_action_policy_libero.sh) and (B) libero-all equal 4-suite mix
(launch_sft_action_policy_libero_all.sh + action_policy_libero_all_repro.toml).
README documents both.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
@fwd4 fwd4 requested review from lfengad, mli0603 and ychao-nvidia July 1, 2026 10:05
@lfengad lfengad merged commit 499bf4d into NVIDIA:main Jul 2, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants